import os
import shutil

# 指定要处理的文件路径
input_directory = '/media/chenglei3/77D014CED257D1411/images/vcg/test'  # 替换为你的文件夹路径

# 遍历目录中的所有文件
for filename in os.listdir(input_directory):
    if filename.endswith('.jpg'):  # 仅处理JPG文件
        # 获取下划线前的部分
        prefix = filename.split('_')[0]

        # 创建新的文件夹路径
        new_folder = os.path.join(input_directory, prefix)

        # 如果文件夹不存在，则创建它
        if not os.path.exists(new_folder):
            os.makedirs(new_folder)

        # 源文件的完整路径
        source_file = os.path.join(input_directory, filename)

        # 目标文件的完整路径
        destination_file = os.path.join(new_folder, filename)

        # 移动文件到新文件夹
        shutil.move(source_file, destination_file)

print("文件分类完成！")
